tensor([[0.9600, 0.5300, 0.0700, 0.7300, 0.5300],
        [0.0400, 0.4700, 0.9300, 0.2700, 0.4700]], grad_fn=<RoundBackward1>)